Added styling and logic for other terrain types