Added in some fun code