Add type to main