removed comments, added return annotation