added some visual fixes