Working on the brute force part