got basic code working