Cleaned up some comments and readme