bugs fixed!