Added some additional functionality to the album controller