Ensure code works