Simplified code, still not working