Added comments and finished if and else statements