Add function to get next keyword