Small fixes

+ Added comments and reduced some code