#9 Add success message