Updated Readme and cleaned up some of the code - imports etc.