2: make main screen similar to the design