made script nicer