Add styling
Add drawing of graphs