updated with my code
Merge https://github.com/veecos/ProgrammingAssignment2