Add caption package to adjust caption width

The caption width were adjusted in order to distinguish them from the regular
text.