added first draft of code