Add define guard