more minor bug fixed
making code nicer