add documentation back in