Square Every Element In Array Python