initial hw2 attempt