Added the initial code I wrote.