add collision function back