fixed a few bugs
adding stuff