Fix code format
Add very basic tests