Wrote comments
Added return statement to exit function early