Clean up if statements.