Minor changes to setup file