Fixed some compiler warnings in problem 1