updated example
clean code