fixed a coiple small issues