Made some changes in Album model