I hope I didn't break everything

Made it draw based on array in test