new solution with take, drop and zip