Tweaks
Added mini test