Fixing dem minor bugs