small modification for better look