Changed file structure. Added setup() function