ui tweaks and fixing image load bug