Add check_position function