Add next part of hw4