My first coding attempt