worked on code style a little