Cleaned up program comments and added to readme TODO