changes to entities class