review of hw1

Added a couple lines to make it functional