need to fix a few bugs