Some styling changes and addition of more documentation