added first round of code