Added methods to models