Checking code against solution