small issues fixed