Made small error fixes