updated a the comments to look cleaner TODO: create a better readme