Tiny commit, just cleaned up some code