Added readme and cleaned code a little