Improved style

Added comments and removed unnecessary whitespaces