Added return annotations