Small update to setup file.