minor bug fixed
minor changes