Fix the documentation

Adding mock modules and fixing documentation style.