Add all needed functions for that