Adjusted readme, and a tiny bit of code