Small change to setup, removed rounds statement