Cook, who played eight games and scored once during his loan spell, feels he's sharpened his skills during his stay in the Midlands.
"Overall it went really well," he told safc.com.
"The way I saw it when I went, it was a win-win situation. I wanted to help Walsall stay up and they achieved that.
"I feel like going there has definitely improved me as a player. Obviously I want to play at the Premier League level but there are still good players and teams in League One.
"Teams can play physical when they want to but they can also pass the ball as well.
"I also had the benefit of playing with experienced players like [goalkeeper] Jimmy Walker and Jonathan Macken, who were always helping me."