add spiral matrix 2 python version