Little update

some bugs fixed