code with dataset