Adding return types for functions