added my first round of code