transforming it into a class