Lab5 code... I already did half of the code in class