Fixed UI issues for small screens. Minor bugs.