Fix bugs, update README