Tiny updates on code