improved doctrings