Fixed slice but not the pythonic way
small error