Added new IF statements