Cleaned up script to be more readable with less duplication, also corrected $ colour to white instead of yellow