Added my updates to the code