proofreading and made it simpler

skipped unnecessary complications (made for better performance)
and added return statements