Add unit test + necesarry implementation to compile. 

Real implementation is missing for now.