Updated Code

just updating the code that I have writen