More Pythonic spike visualisation implementation on TraceVeiw