Making get_albums() more flexible