Update readme
Separate class from main program