Adding proof of concept code and imports