I just added a code

This is an update to what I worked on last night :)