Updated readme and fixed bug.