Added skelet code from notebook