Here's the initial code