Update working prototype