prob2 completely done