changed the code for deployment