fixed bugs yeaah