Add code with some minor changes