Fixing style on function