Changes to the required code to work.