Added comments
Updated styling