How do I do that? Is my previous code correct?