HW5: minor changes

Signed-off-by: unknown <zkk12345@gmail.com>