updated the definition of the problem
make python starting code smaller