Improves UI for iPhone
Fixing desktop version