Fixed styling issues + documentation