End of class

Couple of if/else statements.