Refactored some stuff, made the code prettier.
Replaced the screenshot with a better one.