Ej1 Update

Small update of the Ej1 exercise after review #1.