Refactor so colors are passed into initializer instead of changed after. Still need to change demo