wrapping it in a class