Updated solution to last weeks lab