return type in function annotations also works now