Small fixes and updated readme