Peg Puzzle in pythom