Small changes

just fixing things up