Made some changes to make the code compile on my machine.