Added the end statements for the two new if statmeents

Did not know ends were required on one-liner ifs. Fixed.