Trying to Solve the Extra Credit Class for the lab