Add methods to album model