Bug fixing but still not working