fixed a few little issues