pretty up the code!