Add box to plots