basic version
v1 sol works for basic case