Adding function prototypes for different state branches