added 
changes to the code