Updated readme and notebook