Array Challenge Coderbyte Solution Python