Update the mockup
Update TODO