Production of quarkonium and heavy flavour in ATLAS