add if else then statements