addressing the feedback for ws_loader.rs file