Make code production ready