s01 comments and code update