Added screenshots, cleaned up code.