Added unit test examples, but entities are not working right now