Simplifying the first exercise