Adding first draft of code