Add if statements and blocks