Adding comments and cleaned up some bad code written for the paper. some more refactoring to do.