more polish on hw3