Use return instead of exit()