Update to prototype

Added more stuff