Make the code work first. Then clean it up. The more you delay cleaning your code up, the more tech debt your project will accumulate.