Fix formatting error
Simplify script a little