Updated lab code and README.md