Made changes to make test pass for Add an Album