Small polishing

Changed to match the final app