lab4 part B ...

Signed-off-by: Quan Xu <quan.xu@intel.com>