Wrapping functions into class