fix small errors, can be implimented later