Still need more if statements