bug fixed

Code runs properly