Add ensure statement and comment