Attempted to add setcolor function and made tweaks