prettier code (i think?)