fixed units problem with ws3 problem 2, woops