Add hit/stay logic