Refactor the parameter being passed into SnakeUI