Tidy Up + Bug Fixes