Not sure if this is right... Added a simple function...