Added wireframe code for TDD