added code to setup