Adding in the code as I have it so far