Changes to diagram, a few updates to problem statement.