Fixed: Include the necessary methods