Minor changes

Just added some minor changes