- make the classes more flexible