Include AssertJ. Closes #2