Minor code cleanup
Updated readme for finished code