minor changes to the setup