Small tweaks and updated readme