Python Square Each Element In Array