worked on the code