Finished the quick prototype