Add Logger and Error Check？