Clean up doc block