Revision to code lab 13