Updated it to run in production