fixing prob1 for exam