Fixed bug, updated readme