making code better? perhaps?
slight changes to code made