Modifying Array In Python