Add post main funcitonality