Add a puzzle selection screen