Adding first draft of code to git repo