Finished code, updated readme