Working on deployment for prod