Finish up the add_border function