Updates test based on style suggestions