tweaks
Add tests