This code is better