Fixed missing function; Code style