Worked on the code..