First simple but working code