Add some more code and tests.

Signed-off-by: Andras Mitzki <mitzkia@gmail.com>