Updates to get working in prod, minor style change
Merge pull request #6 from ofagbemi/prod

Updates to get working in prod, minor style change