you can simplify the first part as well