Put initState in correct place