Add tests with graphics :)