Tiny updates

A couple minor updates in the code