Add function to determine server type