remvoed some bugs