css improvement + better sizing