FIrst working version
Add brute force test