Update

updated to match requirements of problem :D