code correction
clean code