Small improvement on tutorial