My first attempt at code