Revising the code to look exactly as it looks on production.