Python Square Of A List