Fixed drawing bug for problem 1