Improve attachment ui