Adding comments and cleaning  up the code to make it easier to collaborate on this project