Pretty major mistake at the end there but this will do as a proof of concept for now