updated the python code so that it would do standard deviation