Two Element Array Python Practice