convolution.py.numpy is useful for handling vectors and matrices.main() should return the object of type pyplot used for plotting. For example:
import matplotlib.pyplot as plt
import numpy as np
...
def main():
fig = plt.figure(figsize=(18, 9))
...
return fig
if __name__ == "__main__":
main()