Better code

Yeah