Small changes

will fix later