Add small screen and medium screen support. Add controls to the screen. Fix #388