Update script to include album in the output file