Fixed the solution, ready to submit.