Basic functionality done

Still need styling