Add readme, fix minor bug