add tests, fix css