Fixes #3
Added return statements to the code to help peopl more used to seeing them