Updated readme and changed code a bit