hw3 update

add plot functionality