Initial solution to assignment without comments
Added file and function comments