Functions for visualization This repo contains all the necessary python visualization code represented as function