added docs and did some styling