Fix modal & add close button