Finally created a basic version that works