add readme, refactor just a little bit of my code