code debugging

testing and fixing little mistake