Fixed layout issue and removed problem5.cpp