def make_album(name,album_name,song_nums = 1):

     dict_album =  {name:[album_name]}

     if int(song_nums) > 1:

         dict_album[name].append(song_nums)

     return dict_album

 first = make_album('Avril','Fly')

 print(str(first)+"\n")

 second = make_album('Jay Zhou','Jay Chou\'s bedside story',10)

 print(str(second)+"\n")

 third = make_album('Tall swift','welcome to New York',12)

 print(str(third)+"\n")

 while True:

     print("\nplease enter the information of album:")

     name = input("singer's name:")

     album_name = input("album's name:")

     song_nums = input("song's number:")

     album = make_album(name,album_name,song_nums)

     print(album)