Changed the script tp be prettier according to Björn's tips.