Try to change the way of defining class