Simplify answer to exercise 5.2

Simplify implementations of take(n) and drop(n)