Cleaned up more and added an example image

Functional code!