small changes to a class