improved code visualization in readme