updated and fixed a lil problem i found