Now basically works, needs to show colours, inferals etc though