First code (practice while watching the Youtube video)