Made the code slightly funnier.