python code for proof of concept..