finished main. passed first test. submit #1... wish me luck