Added comments
Updated docs