Add HTDF Homework 1: Design a function to create a square based on the given colour