Modifications to code