Fix doctrings