Using proper code for problem.