Bug Fix & Tweaks