Added rough draft of code