small corrections and a challenge at the end ;)