Branch assignment2 integration