Python Type Hints For Mini Puzzle Help