need to fix the border