Small changes

:smile: