Add some methods to models