Finished first draft of code.. now to debug