Todo and docs update