Task 5 design update