Added first draft of code