added in cheese function, cleaned up code