Add some lines to the if/else statements