Changed the way we return the best initial solution