Added the first draft of code